import pyinputplus as pyip

menu = ['red','blue','yellow']
# s = pyip.inputNum('请输入数字：')
# print(s)
# user_input = pyip.inputMenu(menu)
# print(user_input)

user_input = pyip.inputChoice(menu)
print(user_input)

user_email = pyip.inputEmail('请输入email')
print(user_email)